import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO('yolo11n.pt')
    model.predict(source='../../datasets/images/test/',
                  imgsz=640,
                  device=[0,1],
                  batch=16,
                  save=True
                  )